package 中等.搜索.深度优先搜索;

import util.TreeNode;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 给你一个二叉树的根结点 root ，请返回出现次数最多的子树元素和。
 * 如果有多个元素出现的次数相同，返回所有出现次数最多的子树元素和（不限顺序）。
 * 一个结点的 「子树元素和」 定义为以该结点为根的二叉树上所有结点的元素之和（包括结点本身）。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/most-frequent-subtree-sum
 */
public class 出现次数最多的子树元素和_508 {

    public static void main(String[] args) {

    }

    /**
     * 深度搜索收集每个 子树元素和 出现的次数，用哈希表存储起来
     * 然后遍历一遍记录出现次数最多的 子树元素和
     *
     * @param root
     * @return
     */
    public int[] findFrequentTreeSum(TreeNode root) {
        Map<Integer, Integer> map = new HashMap<>();
        int maxCount = 0;
        List<Integer> list = new ArrayList<>();
        dfs(root, map);
        for (Integer key : map.keySet()) {
            if (map.get(key) > maxCount) {
                list = new ArrayList<>();
                list.add(key);
                maxCount = map.get(key);
            } else if (map.get(key) == maxCount) {
                list.add(key);
            }
        }

        int[] ans = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            ans[i] = list.get(i);
        }
        return ans;
    }

    private int dfs(TreeNode node, Map<Integer, Integer> map) {
        if (node == null) {
            return 0;
        }
        int curSum = node.val + dfs(node.left, map) + dfs(node.right, map);
        map.put(curSum, map.getOrDefault(curSum, 0) + 1);
        return curSum;
    }
}
